import {ChatPromptTemplate} from "@langchain/core/prompts";
import {GetLocal} from "@/agent/base/llm.ts";

// @ts-ignore
const ChatLLMTest = async () => {

    const llm = await GetLocal()

    const template = ChatPromptTemplate.fromMessages([
        ["system", "你是一个{role}, 使用中文回答用户的问题！"],
        ["human", "用户的问题是：{userInput}"]
    ])

    const resp = await llm.invoke(await template.invoke({
        role: '笑话生成器',
        userInput: '说一个关于狗的笑话,100字以内'
    }))

    console.log(resp)

}

ChatLLMTest().then()